package com.lagou.edu.factory;

import com.alibaba.druid.util.StringUtils;
import com.lagou.edu.annotation.MyAutowired;
import com.lagou.edu.annotation.MyService;
import com.lagou.edu.annotation.MyTransactional;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.reflections.Reflections;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * @author 应癫
 *
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String,Object> map = new HashMap<>();  // 存储对象


    static {
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        // InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
        // // 解析xml
        // SAXReader saxReader = new SAXReader();
        // try {
        //     Document document = saxReader.read(resourceAsStream);
        //     Element rootElement = document.getRootElement();
        //     List<Element> beanList = rootElement.selectNodes("//bean");
        //     for (int i = 0; i < beanList.size(); i++) {
        //         Element element =  beanList.get(i);
        //         // 处理每个bean元素，获取到该元素的id 和 class 属性
        //         String id = element.attributeValue("id");        // accountDao
        //         String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
        //         // 通过反射技术实例化对象
        //         Class<?> aClass = Class.forName(clazz);
        //         Object o = aClass.newInstance();  // 实例化之后的对象
        //
        //         // 存储到map中待用
        //         map.put(id,o);
        //
        //     }
        //
        //     // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
        //     // 有property子元素的bean就有传值需求
        //     List<Element> propertyList = rootElement.selectNodes("//property");
        //     // 解析property，获取父元素
        //     for (int i = 0; i < propertyList.size(); i++) {
        //         Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
        //         String name = element.attributeValue("name");
        //         String ref = element.attributeValue("ref");
        //
        //         // 找到当前需要被处理依赖关系的bean
        //         Element parent = element.getParent();
        //
        //         // 调用父元素对象的反射功能
        //         String parentId = parent.attributeValue("id");
        //         Object parentObject = map.get(parentId);
        //         // 遍历父对象中的所有方法，找到"set" + name
        //         Method[] methods = parentObject.getClass().getMethods();
        //         for (int j = 0; j < methods.length; j++) {
        //             Method method = methods[j];
        //             if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
        //                 method.invoke(parentObject,map.get(ref));
        //             }
        //         }
        //
        //         // 把处理之后的parentObject重新放到map中
        //         map.put(parentId,parentObject);
        //
        //     }
        //
        //
        // } catch (DocumentException e) {
        //     e.printStackTrace();
        // } catch (ClassNotFoundException e) {
        //     e.printStackTrace();
        // } catch (IllegalAccessException e) {
        //     e.printStackTrace();
        // } catch (InstantiationException e) {
        //     e.printStackTrace();
        // } catch (InvocationTargetException e) {
        //     e.printStackTrace();
        // }

        try{
            // 扫描包下的所有类，查找包含MyService注解的类
            Reflections reflections = new Reflections("com.lagou.edu");
            Set<Class<?>> classSet = reflections.getTypesAnnotatedWith(MyService.class);
            // 遍历class判断注解是否有自定义对象id
            for (Class<?> aClass : classSet) {
                Object o = aClass.newInstance();
                MyService myService = aClass.getAnnotation(MyService.class);

                if(StringUtils.isEmpty(myService.value())){
                    //MyService上没有写自定义serviceId的话需要手动获取
                    // 这里需要处理一下name，因为自定义value一般都是直接取当前类的类名，而class.getName拿到的是全路径类名，需要分割获取直接类名
                    String classpathName = aClass.getName();
                    String[] classpathNameArgs = classpathName.split("\\.");
                    String name = classpathNameArgs[classpathNameArgs.length-1];
                    map.put(name, o);
                }else {
                    map.put(myService.value(), o);
                }
            }



            //遍历每个class，查看是否有MyAutowired注解的属性和MyTransactional注解的方法
            for (Map.Entry<String, Object> entry : map.entrySet()) {
                String key = entry.getKey();
                Object classObject = entry.getValue();
                Class<?> aClass = classObject.getClass();

                //获取类中的所有属性并进行遍历
                Field[] classDeclaredFields = aClass.getDeclaredFields();
                for (Field field : classDeclaredFields) {
                    if(field.isAnnotationPresent(MyAutowired.class)){
                        //找到有自定义注解的属性后，查询包含该属性的类中是否有set属性的方法，并通过invoke调用
                        String[] names = field.getType().getName().split("\\.");
                        String name = names[names.length - 1];
                        Method[] methods = aClass.getMethods();
                        for (Method method : methods) {
                            if (method.getName().equalsIgnoreCase("set"+ field.getName())){
                                method.invoke(classObject,map.get(name));
                            }
                        }
                    }
                }

                //遍历类中的方法，查询是否包含@MyTransactional自定义注解
                Method[] aClassMethods = aClass.getDeclaredMethods();
                for (Method aClassMethod : aClassMethods) {
                    if(aClassMethod.isAnnotationPresent(MyTransactional.class)){
                        ProxyFactory proxyFactory = (ProxyFactory) BeanFactory.getBean("proxyFactory");
                        Class<?>[] interfaces = aClass.getInterfaces();
                        if (interfaces != null && interfaces.length > 0){
                            classObject = proxyFactory.getJdkProxy(classObject);
                        }else {
                            classObject = proxyFactory.getCglibProxy(classObject);
                        }
                    }
                }

                map.put(key, classObject);

            }

        }catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }

    }


    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static  Object getBean(String id) {
        return map.get(id);
    }

}
